from flask import Flask, Blueprint, request , current_app
from flask import jsonify
import os
import json
import re
import subprocess

# 创建bp
bp = Blueprint('home', __name__, url_prefix='')


def load_nginx_config():
    # 获取当前脚本的目录
    # current_dir = os.path.dirname(os.path.abspath(__file__))
    # 构建文件的相对路径
    # file_path = os.path.join(current_dir, 'config.json')
    file_path = "F:\\ziliao\\nginx-1.20.1\\conf\\nginx.conf"
    # 打开 JSON 文件并加载数据
    with open(file=file_path, encoding='utf-8', mode='r') as f:
        return f.read()


def update_nginx_config(string_to_write):
    # 文件名
    file_path = "F:\\ziliao\\nginx-1.20.1\\conf\\nginx.conf"

    try:
        with open(file_path, 'w', encoding='utf-8') as file:
            file.write(string_to_write)

        current_app.logger.critical(f"new str is write ok to {file_path} ")
    except Exception as e:
        current_app.logger.critical(f"write file error : {e}")


@bp.get('/')
def index():
    return jsonify(success=True, code=200, msg="hello world")


@bp.post('/register')
def register():
    ip_list = request.get_json(force=True)
    filtered_ips = [ip for ip in ip_list if ip.startswith('172.16.10.')]
    new_ip = filtered_ips[0]
    current_app.logger.critical('program send new ip address is : ' + new_ip)

    nginx_config_str = load_nginx_config()
    # 定义正则表达式模式，用于匹配IP地址中最后一部分可变的情况
    pattern = re.compile(r'172\.16\.10\.\d{1,3}')
    # 原始字符串

    # 使用findall方法找出所有符合正则表达式的内容，并形成列表
    matches = re.findall(pattern, nginx_config_str)
    old_ip = matches[0]
    current_app.logger.critical(f'old ip address in config is : {old_ip}')

    if old_ip != new_ip:
        current_app.logger.critical(f'ip address is change , start update config file ... ')

        # 使用sub方法进行替换，通过lambda函数来构建替换内容
        new_config_str = re.sub(pattern, new_ip, nginx_config_str)
        # print(f'------------------------------------------------新配置文件内容------------------------------------------------')
        # print(new_config_str)
        update_nginx_config(new_config_str)
        current_app.logger.critical(f'config file is update ok , restart nginx')
        exec_command([
            "net stop nginx",
            "net start nginx"
            # "F:",
            # "cd F:\\ziliao\\nginx-1.20.1",
            # "F:\\ziliao\\nginx-1.20.1\\nginx.exe -s reload"
        ])
    else:
        current_app.logger.critical(f'ip address has no change , keep watching...')

    return jsonify(success=True, code=200, msg="register success")


def exec_command(cmd_command_lines):
    # # 多行CMD命令
    # cmd_command_lines = [
    #     "cd C:\\Program Files",
    #     "dir"
    # ]
    # 将多行命令拼接成一个字符串
    cmd_command = " & ".join(cmd_command_lines)

    # 使用subprocess.run()方法执行CMD命令
    result = subprocess.run(cmd_command, shell=True, capture_output=True, text=True)

    # 输出命令执行结果
    if result.returncode == 0:
        current_app.logger.critical("command run success")
        current_app.logger.critical(result.stdout)
    else:
        current_app.logger.critical("command run fail . error message：")
        current_app.logger.critical(result.stderr)
